// Copyright (c) 2010 SuccessFactors, Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//
//     * Redistributions of source code must retain the above
//       copyright notice, this list of conditions and the following
//       disclaimer.
//
//     * Redistributions in binary form must reproduce the above
//       copyright notice, this list of conditions and the following
//       disclaimer in the documentation and/or other materials
//       provided with the distribution.
//
//     * Neither the name of the SuccessFactors, Inc. nor the names of
//       its contributors may be used to endorse or promote products
//       derived from this software without specific prior written
//       permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
// INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
// OF THE POSSIBILITY OF SUCH DAMAGE.

package org.owasp.jxt.compiler;

import java.io.File;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import javax.tools.Diagnostic;
import javax.tools.DiagnosticCollector;
import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;

/**
 * Java6ToolsCompiler -- Uses Java 1.6's javax.tools.JavaCompiler.
 * (Aka JSR 199)
 *
 * @author Jeffrey Ichnowski
 * @version $Revision: 8 $
 */
public class Java6ToolsCompiler extends Compiler {

    private JavaCompiler _compiler;
    private StandardJavaFileManager _fileManager;

    public Java6ToolsCompiler() {
        _compiler = ToolProvider.getSystemJavaCompiler();
        _fileManager = _compiler.getStandardFileManager(null, null, null);
    }

    public final CompileResults compile(String ... classNames) {

        File[] sourceFiles = new File[classNames.length];
        for (int i=0 ; i<classNames.length ; ++i) {
            sourceFiles[i] = classToSourceFile(classNames[i]);
        }

        Iterable<? extends JavaFileObject> compilationUnits =
            _fileManager.getJavaFileObjects(sourceFiles);

        List<String> options = new ArrayList<String>();
        options.add("-Xlint:unchecked");

        if (_debug) {
            options.add("-g");
        }

        if (_source != null) {
            options.add("-source");
            options.add(_source);
        }

        if (_target != null) {
            options.add("-target");
            options.add(_target);
        }

        DiagnosticCollector<JavaFileObject> diagnostics = new DiagnosticCollector<JavaFileObject>();

        Boolean success = _compiler.getTask(null, _fileManager, diagnostics, options, null, compilationUnits).call();
        Locale locale = Locale.getDefault();
        StringWriter detail = new StringWriter();
        PrintWriter out = new PrintWriter(detail);
        for (Diagnostic<? extends JavaFileObject> d : diagnostics.getDiagnostics()) {
            out.println(d.getSource().getName() + ":" +
                        d.getLineNumber() + ":" +
                        d.getColumnNumber() + ": " +
                        d.getMessage(locale));
        }

        return new CompileResults(success, detail.toString());
    }

} // Java6ToolsCompiler
